from .legacy import load_weight_from_legacy_checkpoint